import openai
import os

# 目前需要设置代理才可以访问 api
os.environ["HTTP_PROXY"] = "http://127.0.0.1:7890"
os.environ["HTTPS_PROXY"] = "http://127.0.0.1:7890"
openai.api_key = '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'
prompt = "为什么会有战争"
response = openai.ChatCompletion.create(
    model="gpt-3.5-turbo",  # 使用GPT-3.5模型
    prompt=prompt,
    max_tokens=50,  # 生成的最大标记数
    n=1,  # 生成的响应数量
)

print(response.choices[0].text.strip())